Skip to content

fix: Block path traversal attacks via the deployment API for MLflow-Triton - #8665

Merged
mudit-eng merged 9 commits into
mainfrom
mudita/tri-665-psirt-triton-inference-server-mlflow-plugin-path-traversal
Mar 4, 2026
Merged

fix: Block path traversal attacks via the deployment API for MLflow-Triton#8665
mudit-eng merged 9 commits into
mainfrom
mudita/tri-665-psirt-triton-inference-server-mlflow-plugin-path-traversal

Conversation

@mudit-eng

@mudit-eng mudit-eng commented Feb 18, 2026

Copy link
Copy Markdown
Contributor

What does the PR do?

This code change sanitizes the model name in the MLflow-Triton deployment plugin to prevent path traversal.

Checklist

  • PR title reflects the change and is of format <commit_type>: <Title>
  • Changes are described in the pull request.
  • Related issues are referenced.
  • Populated github labels field
  • Added test plan and verified test passes.
  • Verified that the PR passes existing CI.
  • Verified copyright is correct on all changed files.
  • Added succinct git squash message before merging ref.
  • All template sections are filled out.
  • Optional: Additional screenshots for behavior/output changes with before/after.

Commit Type:

Check the conventional commit type
box here and add the label to the github PR.

  • build
  • ci
  • docs
  • feat
  • fix
  • perf
  • refactor
  • revert
  • style
  • test

Related PRs:

Where should the reviewer start?

Test plan:

Added a unit test and verified locally. Also, tested in CI.

  • CI Pipeline ID:
    44324776, 45253345

Caveats:

Background

Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)

  • closes GitHub issue: #xxx

@mudit-eng mudit-eng changed the title [TRI-665] Block path traversal attacks via the deployment API for MLf… [TRI-665] Block path traversal attacks via the deployment API for MLflow-Triton Feb 18, 2026
@yinggeh

yinggeh commented Feb 18, 2026

Copy link
Copy Markdown
Contributor

Comment thread deploy/mlflow-triton-plugin/mlflow_triton/deployments.py Outdated
Comment thread qa/L0_mlflow/plugin_test.py Outdated
Comment thread qa/L0_mlflow/plugin_test.py Outdated
Comment thread deploy/mlflow-triton-plugin/mlflow_triton/deployments.py Outdated
@mudit-eng mudit-eng changed the title [TRI-665] Block path traversal attacks via the deployment API for MLflow-Triton fix: Block path traversal attacks via the deployment API for MLflow-Triton Feb 19, 2026
@mudit-eng
mudit-eng requested a review from yinggeh February 19, 2026 22:17
if flavor not in self.supported_flavors:
raise Exception("{} model flavor not supported by Triton".format(flavor))

def _validate_model_name(self, name):

@yinggeh yinggeh Feb 20, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just a note no need to fix today, there are multiple similar pattern @whoisj added to address PSIRT issues in server. We could refactor them into one common utility fuction.

Comment thread deploy/mlflow-triton-plugin/mlflow_triton/deployments.py
Comment thread qa/L0_mlflow/plugin_test.py Outdated
Comment thread qa/L0_mlflow/plugin_test.py Outdated
Comment thread qa/L0_mlflow/plugin_test.py Outdated
@mudit-eng
mudit-eng requested a review from yinggeh February 24, 2026 20:55
@mudit-eng
mudit-eng requested a review from pskiran1 February 26, 2026 08:39
Comment thread deploy/mlflow-triton-plugin/mlflow_triton/deployments.py Outdated
Comment thread qa/L0_mlflow/plugin_test.py Outdated
@mudit-eng
mudit-eng requested a review from yinggeh March 3, 2026 20:54

@yinggeh yinggeh left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work

@mudit-eng
mudit-eng merged commit 04af2a6 into main Mar 4, 2026
3 checks passed
@mudit-eng
mudit-eng deleted the mudita/tri-665-psirt-triton-inference-server-mlflow-plugin-path-traversal branch March 4, 2026 17:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Development

Successfully merging this pull request may close these issues.

3 participants